package com.atguigu.search;

import java.util.ArrayList;
import java.util.Arrays;

/**
 * @Author : SunFeng  注意二分查找的前提是 该数组是有序的
 * @Description: 1 首先确定该数组的中间下标 mid=(left+right)/2
 * 2 然后让我查找的值findVal 与arr[mid]比较
 * 2.1 findVal>arr[mid] 说明要查找的值在数组的右边
 * 2.2 findVal<arr[mid] 说明要查找的值在数组的左边
 * 2.3 findVal=arr[mid] 说明找到了要查找的值
 * @Date : 2021/7/8  14:42
 */
public class BinarySearch {
    public static void main(String[] args) {
        int arr[] = {1, 8, 10, 89, 1000, 1000, 1000, 1000, 1234};
        int arr2[] = {1, 2, 2, 2, 3, 4, 5};
        int i = binarySearch(arr, 0, arr.length - 1, 1000);
        System.out.println(i);

    }

    /**
     * @param array   数组
     * @param left    左边的索引
     * @param right   右边的索引
     * @param findVal 要查找的值
     * @return 如果找到就返回下标， 如果没有找到就返回-1
     */
    public static int binarySearch(int[] array, int left, int right, int findVal) {
        //当left>right 时 说明递归整个数组完毕
        if (left > right) {
            return -1;
        }
        int mid = (left + right) / 2;
        int midVal = array[mid];
        if (midVal > findVal) {
            return binarySearch(array, mid + 1, right, findVal);
        } else if (midVal < findVal) {
            return binarySearch(array, left, mid, findVal);
        } else {
            return mid;
        }
    }

    /**
     * @param array   数组
     * @param left    左边的索引
     * @param right   右边的索引
     * @param findVal 要查找的值
     * @return 找到所有的满足条件的元素下标
     */
    public static ArrayList<Integer> binarySearchAll(int[] array, int left, int right, int findVal) {

        // 如果 左边值大于右边值 返回空
        if (left > right) {
            return new ArrayList<>();
        }
        // 找出中间索引
        int mid = (left + right) / 2;
        int midVal = array[mid];

        if (findVal > midVal) {
            return binarySearchAll(array, mid + 1, right, findVal);
        } else if (findVal < midVal) {
            return binarySearchAll(array, left, mid - 1, findVal);
        } else {

            int temp = mid - 1;
            ArrayList<Integer> arrayList = new ArrayList<>();
            while (temp >= 0 && findVal == array[temp]) {
                arrayList.add(temp);
                temp--;
            }
            arrayList.add(mid);
            temp = mid + 1;
            while (temp <= array.length - 1 && findVal == array[temp]) {
                arrayList.add(temp);
                temp++;
            }
            return arrayList;

        }

    }
}
